package com.demo.algorithm.tradeStock06;

/**
 * Author:       zhangshaoyang
 * Description: k = +infinity with fee, 每次交易(买卖一次算一次交易)要支付手续费，只要把手续费从利润中减去即可。
 * 给定一个整数数组 prices，其中第 i 个元素代表了第 i 天的股票价格 ；整数 fee 代表了交易股票的手续费用。
 * <p>
 * 你可以无限次地完成交易，但是你每笔交易都需要付手续费。如果你已经购买了一个股票，在卖出它之前你就不能再继续购买股票了。
 * <p>
 * 返回获得利润的最大值。
 * <p>
 * 注意：这里的一笔交易指买入持有并卖出股票的整个过程，每笔交易你只需要为支付一次手续费。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：prices = [1, 3, 2, 8, 4, 9], fee = 2
 * 输出：8
 * 解释：能够达到的最大利润:
 * 在此处买入 prices[0] = 1
 * 在此处卖出 prices[3] = 8
 * 在此处买入 prices[4] = 4
 * 在此处卖出 prices[5] = 9
 * 总利润: ((8 - 1) - 2) + ((9 - 4) - 2) = 8
 * 示例 2：
 * <p>
 * 输入：prices = [1,3,7,5,10,3], fee = 3
 * 输出：6
 * <p>
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-transaction-fee
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * Date:         2021/7/13 6:14 下午
 */
public class KInfinityWithFee {
    public static int KInfinityWithFee(int[] prices, int fee) {
//        dp[i][0] = max(dp[i-1][0], dp[i-1][1] + prices[i])
//        dp[i][1] = max(dp[i-1][1], dp[i-1][0] - prices[i] - fee)
//        解释：相当于买入股票的价格升高了。
//        在第一个式子里减也是一样的，相当于卖出股票的价格减小了。不过在第一个式子里减要考虑i为0时候的情况。

        int n = prices.length;
        int dp_i_0 = 0, dp_i_1 = Integer.MIN_VALUE;
        for (int i = 0; i < n; i++) {
//            if (i == 0) {
//                dp_i_0 = 0;
//                dp_i_1 = -prices[0];
//                continue;
//            }
//            dp_i_0 = Math.max(dp_i_0, dp_i_1 + prices[i] - fee);
//            dp_i_1 = Math.max(dp_i_1, dp_i_0 - prices[i]);

            dp_i_0 = Math.max(dp_i_0, dp_i_1 + prices[i]);
            dp_i_1 = Math.max(dp_i_1, dp_i_0 - prices[i] - fee);
        }
        return dp_i_0;
    }

    /**
     * 最终代码
     *
     * @param prices
     * @param fee
     * @return
     */
    public static int KInfinityWithFee1(int[] prices, int fee) {
       int n = prices.length;
        int dp_i_0 = 0, dp_i_1 = Integer.MIN_VALUE;
        for (int i = 0; i < n; i++) {
            dp_i_0 = Math.max(dp_i_0, dp_i_1 + prices[i]);
            dp_i_1 = Math.max(dp_i_1, dp_i_0 - prices[i] - fee);
        }
        return dp_i_0;
    }

    public static void main(String[] args) {
        int[] prices = {1, 3, 2, 8, 4, 9};
        int[] prices1 = {1, 3, 7, 5, 10, 3};
        int i = KInfinityWithFee1(prices, 2);
        int i1 = KInfinityWithFee1(prices1, 3);
        System.out.println(i);
        System.out.println(i1);
    }
}
